'''
实验题目：写一个批量重命名脚本
需求：1.接受用户指定的前缀；
    2.接受用户指定的路径，如果不存在，则报错；
    3.接受用户指定的操作，1.增加，2.删除；
    4.不考虑报错，新增加，再删除前缀。
'''
import os
while True:
    flag=input('请输入您需要的批量重命名文件操作，1.添加前缀，2.删除前缀，3.退出：')
    if flag=='3':
        print('接受到退出信息，已退出。')
        break
    elif flag=='1':
        prefix=input('请输入您希望的前缀：')
        path=input('请输入您希望批量重命名的文件夹(请填写绝对路径)：')
        os.chdir(path)
        filename=os.listdir('./')
        for i in filename:
            newname=prefix+i
            os.rename(i,newname)
        print(os.listdir())
    elif flag=='2':
        prefix=input('请输入需要删除的前缀：')
        path=input('请输入您希望批量重命名的文件夹(请填写绝对路径)：')
        os.chdir(path)
        filename=os.listdir('./')
        for i in filename:
            #路径的文件拼接还是有问题，可以优化
            index=i.find(prefix)+len(prefix)
            newname=i[index:]
            os.rename(i,newname)
        print(os.listdir())
    else:
        print('您输入有误，请重新输入：')